import walf
import funksvd
import pandas as pd
import matplotlib
matplotlib.use('TkAgg')
import matplotlib.pyplot as plt
import os
import numpy as np
from getFuncton import get_train


x=list(range(50,90,10));
y=[]
z=[]
a=[]
b=[]
c=[]
data=pd.read_csv("ratings.csv",sep="\s+")
# for i in range(5,9):
all = get_train(data,70)
train = all[0]
test = all[1]
for j in range(20,60,10):
    y.append(funksvd.funksvd(train,test,j))
    z.append(walf.walf(train,test,0.2,j))
    a.append(walf.walf2(train, test,j))
    b.append(walf.walf1(train, test,j))







print(y,z,a,b)